//#include "stdio.h"        // C++中建议使用C++标准库而非C库
#include <sys/socket.h>     // 包含socket()、bind()、connect()等函数声明
#include <netinet/in.h>     // 包含AF_INET、SOCK_STREAM等常量定义
#include <arpa/inet.h>      // 包含inet_addr()等IP地址处理函数（如果需要）
#include <sys/epoll.h>      // 包含epoll()
#include <unistd.h>         // 包含close()函数

#include <cstdio>
#include <cstring>          // 提供std::strerror
#include <string>
#include <vector>
using std::string;
using std::vector;

#define MAX_EVENTS 10
#define BUFFER_SIZE 1024

inline bool ret_check(int ret, const string& msg = "", bool should_exit = true)
{ 
    if(ret < 0){
        char message[256];

        snprintf(message, sizeof(message), 
                 "%s | %d | %s | error : %s\n | errno = %d(%s) \n", 
                 (const char*)__FILE__, __LINE__, __FUNCTION__, msg.c_str(), errno, std::strerror(errno));
        perror(message);

        if(should_exit)
        {
            exit(EXIT_FAILURE);
        }
        return false;
    } 
    return true;
}

bool test(){
    int ret = 0;

    // 1. 创建监听套接字
    int listen_fd = socket(AF_INET, SOCK_STREAM, 0);
    ret_check(listen_fd, "socket failed");

    // 设置地址复用（避免TIME_WAIT 状态导致绑定失败）
    int opt = 1;
    // 套接字秒附属 | 选项所属的协议层 | 具体的选项名称 | 指向选项值的指针 | 选项值的长度
    // SOL_SOCKET 是通用套接字选项（最常用）,影响套接字本身的行为，与具体协议无关
    setsockopt(listen_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));

    // 2. 准备地址结构 & 绑定地址
    struct sockaddr_in server_addr = {0};
    server_addr.sin_family = AF_INET;
    //server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");   // 监听特定IP
    server_addr.sin_addr.s_addr = INADDR_ANY;             // 监听所有接口（推荐）
    server_addr.sin_port = htons(8080);

    // 注意结构体名称，是sock_addr*而不是sockaddr
    ret = bind(listen_fd, (struct sockaddr*)&server_addr, sizeof(server_addr));
    ret_check(ret, "bind failed");

    // 3. 监听连接
    ret = listen(listen_fd, 5);
    ret_check(ret, "listen failed");

    printf("Server listening on port 8080...\n");

    // 4. 创建epoll实例（在while外调用一次）
    int epoll_fd = epoll_create(1);
    ret_check(epoll_fd, "epoll_create failed");

    // 5. 将监听套接字加入epoll监控（监听读事件）
    epoll_event ev;
    ev.events = EPOLLIN;    // 监听读事件（新链接到来）
    ev.data.fd = listen_fd;
    ret = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, listen_fd, &ev);
    ret_check(ret, "epoll_ctl");

    // 存储就绪队列的数据
    vector<epoll_event> events(MAX_EVENTS);

    while(true)
    {
        int nfds = epoll_wait(epoll_fd, events.data(), MAX_EVENTS, -1);
        if(nfds == -1){
            perror("epoll_wait");
            continue;
        }

        for(int i = 0; i < nfds; ++i){

            if(events[i].data.fd == listen_fd)
            {
                // 4. [关键缺失]必须添加accept()循环，否则服务器会立即退出
                struct sockaddr_in client_addr;
                socklen_t client_len = sizeof(client_addr);
                int client_fd = accept(listen_fd, (struct sockaddr*)&client_addr, &client_len);
                if(client_fd == -1)
                {
                    perror("accept");
                    continue;
                }

                // 处理客户端连接
                printf("Client connected: %s:%d\n", 
                       inet_ntoa(client_addr.sin_addr),
                       ntohs(client_addr.sin_port));

                // 将新客户端套接字加入epoll监控（默认监听读事件）
                ev.events = EPOLLIN | EPOLLET;  // 使用边缘触发模式（ET）
                ev.data.fd = client_fd;
                if(epoll_ctl(epoll_fd, EPOLL_CTL_ADD, client_fd, &ev) == -1)
                {
                    perror("epoll_ctl: client_fd");
                    close(client_fd);
                }
            }

            else if(events[i].events & EPOLLIN)
            {
                int client_fd = events[i].data.fd;
                char buffer[BUFFER_SIZE] = {0};
                int n = read(client_fd, buffer, BUFFER_SIZE);

                if (n <= 0) {
                    // 客户端关闭连接或发生错误
                    printf("Client disconnected\n");
                    close(client_fd);
                    epoll_ctl(epoll_fd, EPOLL_CTL_DEL, client_fd, nullptr);
                } else {
                    printf("Received: %s\n", buffer);

                    // 【关键】动态调整事件：读完数据后，改为监听写事件（准备回显数据）
                    ev.events = EPOLLOUT | EPOLLET;
                    ev.data.fd = client_fd;
                    epoll_ctl(epoll_fd, EPOLL_CTL_MOD, client_fd, &ev);
                }

            }

            // 4.3 处理客户端数据可写事件
            else if (events[i].events & EPOLLOUT) {
                int client_fd = events[i].data.fd;
                const char* response = "Server received your message!";
                write(client_fd, response, strlen(response));

                // 【关键】写完后，改回监听读事件（等待下一次数据）
                ev.events = EPOLLIN | EPOLLET;
                ev.data.fd = client_fd;
                epoll_ctl(epoll_fd, EPOLL_CTL_MOD, client_fd, &ev);
            }
        }
    }

    close(listen_fd);
    close(epoll_fd);
    return 0;
}

int main()
{
    test();
    return 0;
}

